package yao.common.excel;



import java.io.BufferedInputStream;
import java.io.BufferedOutputStream;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.CellStyle;
import org.apache.poi.ss.usermodel.Font;
import org.apache.poi.ss.usermodel.IndexedColors;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;


/**
 * 
 * @author chenyao
 * @date 2016年8月15日 上午10:35:38
 */
public class ExcelCreator {
	private enum Version {
		EXCEL_2003(".xls") {
			public Workbook createWorkbook() {
				return new HSSFWorkbook();
			}
		}, 
		EXCEL_2007(".xlsx") {
			public Workbook createWorkbook() {
				return new XSSFWorkbook();
			}
		};
		private final String suffix;		
		private Version(String suffix) {
			this.suffix = suffix;
		}

		public String getSuffix() {
			return suffix;
		}
		public abstract Workbook createWorkbook();
		
	}
	
	public static ExcelCreator createExcel2003(String path, String fileName) {
		return new ExcelCreator(path, fileName, Version.EXCEL_2003);
	}
	
	public static ExcelCreator createExcel2007(String path, String fileName) {
		return new ExcelCreator(path, fileName, Version.EXCEL_2007);
	}
	
	private final String path;	
	private final String fileName;
	private final Version version;
	private ExcelCreator(String path, String fileName, Version version) {
		super();
		this.path = path;
		this.fileName = fileName;
		this.version = version;
	}

	/**
	 * 下载excel文件
	 * @param excelDownload
	 * @author chenyao
	 * @date 2016年8月15日 下午2:57:27
	 */
	public void download(ExcelData excelData) {
		InputStream inputStream = getExportedData(excelData.getContent(),
				excelData.getKeys(), excelData.getColumnNames());
		try {
			setExcelResponse(inputStream);
		} catch (Exception e) {
			e.printStackTrace();
		}
	}
	
	
	/**
	 * 创建excel文档，
	 * 
	 * @param list
	 *            数据
	 * @param keys
	 *            list中map的key数组集合
	 * @param columnNames
	 *            excel的列名
	 */
	private Workbook createWorkBook(List<Map<String, Object>> list, String[] keys, String columnNames[]) {
		
		//添加sheetName//
		Map<String, Object> sheetNameMap = new HashMap<String, Object>();
		sheetNameMap.put("sheetName", "sheet1");
		list.add(0, sheetNameMap);
		
		// 创建excel工作簿
		Workbook wb = this.version.createWorkbook();
		// 创建第一个sheet（页），并命名
		Sheet sheet = wb.createSheet((String) list.get(0).get("sheetName"));
		// 手动设置列宽。第一个参数表示要为第几列设；，第二个参数表示列的宽度，n为列高的像素数。
		for (int i = 0; i < keys.length; i++) {
			sheet.setColumnWidth((short) i, (short) (35.7 * 150));
		}

		// 创建第一行
		Row row = sheet.createRow((short) 0);

		// 创建两种单元格格式
		CellStyle cs = wb.createCellStyle();
		CellStyle cs2 = wb.createCellStyle();

		// 创建两种字体
		Font f = wb.createFont();
		Font f2 = wb.createFont();

		// 创建第一种字体样式（用于列名）
		f.setFontHeightInPoints((short) 10);
		f.setColor(IndexedColors.BLACK.getIndex());
		f.setBoldweight(Font.BOLDWEIGHT_BOLD);

		// 创建第二种字体样式（用于值）
		f2.setFontHeightInPoints((short) 10);
		f2.setColor(IndexedColors.BLACK.getIndex());

		// Font f3=wb.createFont();
		// f3.setFontHeightInPoints((short) 10);
		// f3.setColor(IndexedColors.RED.getIndex());

		// 设置第一种单元格的样式（用于列名）
		cs.setFont(f);
		cs.setBorderLeft(CellStyle.BORDER_THIN);
		cs.setBorderRight(CellStyle.BORDER_THIN);
		cs.setBorderTop(CellStyle.BORDER_THIN);
		cs.setBorderBottom(CellStyle.BORDER_THIN);
		cs.setAlignment(CellStyle.ALIGN_CENTER);

		// 设置第二种单元格的样式（用于值）
		cs2.setFont(f2);
		cs2.setBorderLeft(CellStyle.BORDER_THIN);
		cs2.setBorderRight(CellStyle.BORDER_THIN);
		cs2.setBorderTop(CellStyle.BORDER_THIN);
		cs2.setBorderBottom(CellStyle.BORDER_THIN);
		cs2.setAlignment(CellStyle.ALIGN_CENTER);
		// 设置列名
		for (int i = 0; i < columnNames.length; i++) {
			Cell cell = row.createCell(i);
			cell.setCellValue(columnNames[i]);
			cell.setCellStyle(cs);
		}
		// 设置每行每列的值
		for (short i = 1; i < list.size(); i++) {
			// Row 行,Cell 方格 , Row 和 Cell 都是从0开始计数的
			// 创建一行，在页sheet上
			Row row1 = sheet.createRow((short) i);
			// 在row行上创建一个方格
			for (short j = 0; j < keys.length; j++) {
				Cell cell = row1.createCell(j);
				String val = "";
				if(list.get(i) != null) {
					val = (String) list.get(i).get(keys[j]);
				}				
				cell.setCellValue(val);
				cell.setCellStyle(cs2);
			}
		}
		return wb;
	}
	
	
	private InputStream getExportedData(List<Map<String, Object>> list, String[] keys, String columnNames[]) {
		ByteArrayOutputStream os = new ByteArrayOutputStream();
		try {
			this.createWorkBook(list, keys, columnNames).write(os);			
		} catch (IOException e) {
			throw new RuntimeException("read data error!");
		}
		byte[] content = os.toByteArray();
		InputStream is = new ByteArrayInputStream(content);		
		return is;
	}

	private void setExcelResponse(InputStream contentInput) throws Exception {
		BufferedInputStream bis = null;
		BufferedOutputStream bos = null;
		try {
			bis = new BufferedInputStream(contentInput);
			String path = this.path + this.fileName + this.version.getSuffix();
			bos = new BufferedOutputStream(new FileOutputStream(new File(path)));
			byte[] buff = new byte[2048];
			int bytesRead;
			while (-1 != (bytesRead = bis.read(buff, 0, buff.length))) {
				bos.write(buff, 0, bytesRead);
			}
		} catch (final IOException e) {
			throw e;
		} finally {
			if (bis != null)
				bis.close();
			if (bos != null)
				bos.close();
		}
	}
	
}